package homework4;

import java.io.*;
import java.net.Socket;
import java.util.Scanner;

/**
 * 4. 编程题
 * 使用基于 tcp 协议的编程模型实现将 UserMessage 类型对象由客户端发送给服务器；
 * 服务器接收到对象后判断用户对象信息是否为 "admin" 和 "123456"，
 * 若是则将UserMessage对象中的类型改为"success"，
 * 否则将类型改为"fail"并回发给客户端，客户端接收到服务器发来的对象后判断并给出登录成功或者失败的提示。
 * 其中 UserMessage 类的特征有：类型(字符串类型) 和 用户对象(User 类型)。
 * 其中 User 类的特征有：用户名、密码(字符串类型)。
 * 如：
 * UserMessage tum = new UserMessage("check", new User("admin", "123456"));
 */
public class ClientTest {

	public static void main(String[] args) {

		Socket socket = null;
		ObjectInputStream objectInputStream = null;
		ObjectOutputStream oos = null;
		Scanner scanner = null;
		try {
			// 1.创建Socket类型的对象与服务器进行链接
			socket = new Socket("127.0.0.1", 8888);
			System.out.println("连接成功！进行校验...");
			oos = new ObjectOutputStream(socket.getOutputStream());
			objectInputStream = new ObjectInputStream(socket.getInputStream());
			scanner = new Scanner(System.in);
			// 2.使用输入输出流进行通信
			while (true) {
				System.out.println("请输入用户名(输入bye结束)");
				String username = scanner.next();
				System.out.println("请输入密码(输入bye结束)");
				String password = scanner.next();
				oos.writeObject(new UserMessage("check", new User(username, password)));
				System.out.println("客户端发送内容成功！");
				if ("bye".equalsIgnoreCase(username) || "bye".equalsIgnoreCase(password)) {
					System.out.println("结束聊天");
					break;
				}
				//获取服务端返回的数据并打印
				Object check = objectInputStream.readObject();
				System.out.println("服务器返回的消息是：" + check);
			}
		} catch (IOException | ClassNotFoundException e) {
			e.printStackTrace();
		} finally {
			if (null != scanner) {
				scanner.close();
			}
			try {
				if (null != oos) {
					oos.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
			try {
				if (null != objectInputStream) {
					objectInputStream.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
			try {
				if (null != socket) {
					socket.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
}
